package com.leetcode.www;

import java.util.ArrayList;
import java.util.List;

/**
 * Creared with IntelliJ IDEA.
 * Description:n皇后问题 研究的是如何将 n个皇后放置在 n × n 的棋盘上，并且使皇后彼此之间不能相互攻击。
 *
 * 给你一个整数 n ，返回 n 皇后问题 不同的解决方案的数量。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/n-queens-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * User:yxd
 * Date:2022-03-17
 * Time:17:41
 */
public class DFS52 {

    public int totalNQueens(int n) {
        if(n <= 0)return 0;
        List<List<pair>> ret = new ArrayList<>();
        List<pair> list = new ArrayList<>();//坐标矩阵
        DFS(ret,list,0,n);
        return ret.size();
    }
    public void DFS(List<List<pair>> ret,List<pair> list,int curRow,int n){
        //边界条件
        if(curRow == n){
            ret.add(new ArrayList(list));
            return;
        }
        for(int i = 0;i < n;i ++){//每行可存放的位置
            if(isTrue(list,curRow,i)){//这个位置是否满足皇后问题
                list.add(new pair(curRow,i));
                DFS(ret,list,curRow + 1,n);
                list.remove(list.size() - 1);//回溯
            }
        }
    }
    public boolean isTrue(List<pair> list,int row,int col){
        for(pair p: list){
            if(p.y == col || ((Math.abs(p.x - row)) == Math.abs(p.y - col))){//列不能相同且不能在同一斜线
                return false;
            }
        }
        return true;
    }
}
